#library(vtree)
source("C:/Users/nickb/Documents/vtree/source.R")
library(dplyr)
library(haven)

data <- read_spss("Viral_Communication_Phase_I-III.sav")

#data <- read_spss("https://zenodo.org/record/5779516/files/Viral_Communication_Phase_I-III.sav")

d <- as_factor(data)

#View(Hmisc::label(d))

agreement <- function(x,neutralAgree=TRUE,agree="Agree",disagree="Disagree") {
  neutral <- if (neutralAgree) agree else disagree
  case_when(
    {{x}} == "Strongly Agree"    ~ agree,
    {{x}} == "Agree"             ~ agree,
    {{x}} == "Somewhat Agree"    ~ agree,
    {{x}} == "Neutral"           ~ neutral,
    {{x}} == "Somewhat Disagree" ~ disagree,
    {{x}} == "Disagree"          ~ disagree,
    {{x}} == "Strongly Disagree" ~ disagree)
}

d <- d %>%
  mutate(
    Phase_1=M_PHASE1_COMPETION,
    Phase_2=M_PHASE2_COMPLETION,
    Phase_3=M_PHASE3_COMPLETION,
    politics_1=factor(case_when(
      SD_POL_ORIENTATION == "Left"   ~ "Left",
      SD_POL_ORIENTATION == "-2"     ~ "Left",
      SD_POL_ORIENTATION == "-1"     ~ "Left",
      SD_POL_ORIENTATION == "Centre" ~ "Centre",
      SD_POL_ORIENTATION == "1"      ~ "Right",
      SD_POL_ORIENTATION == "2"      ~ "Right",
      SD_POL_ORIENTATION == "Right"  ~ "Right"),levels=c("Left","Centre","Right")),
    mask_1=case_when(
      PHASE1_AC_EFF_MASK == "Extremely effective"            ~ "Effective",
      PHASE1_AC_EFF_MASK == "Very effective"                 ~ "Effective",
      PHASE1_AC_EFF_MASK == "Effective"                      ~ "Effective",
      PHASE1_AC_EFF_MASK == "Somewhat effective"             ~ "Less effective",
      PHASE1_AC_EFF_MASK == "Not effective at all effective" ~ "Less effective"),
    world_1=agreement(PHASE1_AS_WORLD,neutralAgree=TRUE),
    science_eb_1=case_when(
      PHASE1_AS_BOR_EXC == "Exciting" ~ "Exciting",
      PHASE1_AS_BOR_EXC == "2"        ~ "Exciting",
      PHASE1_AS_BOR_EXC == "1"        ~ "Exciting",
      PHASE1_AS_BOR_EXC == "0"        ~ "Not Exciting",
      PHASE1_AS_BOR_EXC == "-1"       ~ "Not Exciting",
      PHASE1_AS_BOR_EXC == "-2"       ~ "Not Exciting",
      PHASE1_AS_BOR_EXC == "Boring"   ~ "Not Exciting"),
    harmless_1=agreement(PHASE1_RA_HARMLESS_r,neutralAgree=TRUE,
      agree="Harmless",disagree="Not Harmless"),
    finance_1=agreement(PHASE1_RA_FINANCE,neutralAgree=FALSE),
    economy_1=agreement(PHASE1_RA_ECONOMY,neutralAgree=FALSE))

u <- d %>% filter(M_PHASE1_COMPETION=="Yes")

lv <- c(
  science_eb_1="Science is",
  harmless_1="Covid is",
  world_1="Science is making the world a better place")

vtree(d,"Phase_1 Phase_2 Phase_3",pattern=TRUE,showlegend=TRUE)

vtree(d,"Phase_1 SD_GENDER",labelvar=c(Phase_1="Completed Phase 1",SD_GENDER="Gender"),splitwidth=Inf)

vtree(u,"science_eb_1",labelvar=c("science_eb_1"="Science is"))

vtree(u,"PHASE1_AS_BOR_EXC=Exciting PHASE1_AC_EFF_MASK")

vtree(u,"science_eb_1",labelvar=c("science_eb_1"="I think science is"))

vtree(u,"science_eb_1 harmless_1 mask_1",
  labelvar=c(science_eb_1="Science is",harmless_1="Covid is"),pattern=TRUE)

vtree(u,"politics_1",splitwidth=Inf,labelvar=c(politics_1="Political orientation"),
  summary=c("harmless_1=Harmless \nCovid is harmless: %pct%","mask_1=Effective \nMasks are effective: %pct%"))

vtree(u,"SD_GENDER politics_1",splitwidth=Inf,
  summary=c("harmless_1=Harmless \nCovid is harmless: %pct%","mask_1=Effective \nMasks are effective: %pct%"))

vtree(u,"SD_GENDER world_1",splitwidth=Inf,
  labelvar=lv)

vtree(u,"finance_1",splitwidth=Inf,
  labelvar=lv)

vtree(u,"economy_1",splitwidth=Inf,
  labelvar=lv)

vtree(u,"finance_1 economy_1",splitwidth=Inf,
  labelvar=lv)